assignment 0 done - tests ok