Completed the assignment with my own work