Completed assignment 2