Figured out assignment 1