Assigment 2 - final