Assigment 2 partial code