Started on assignment 5