fix assignment 1