adding code to assignment