Homework #2

Improved script a little bit:D