assignment 1 code