Assignment 1 done :)