Main script

All the code you need for assignment