Completed assignment 5