assignment 2 done and tested