Progress on assignment 5