Kinda finished assignment 1