Advanced Python Code With Multiple Functions