Updated with short explanation of work done for assignment.