Working on assingment 2