Update 2.py

cleaner code