Python Pass User Input To Function Properly The Head