how to write efficent code in python