Completed assignment 4