Fix on the styling to adhere PEP8
Refactored program for the better design
All the if/else logic is now in the separate functions.
Main execution portion of the program is now cleaner and more readable.