Committed the edited and tested solution to the assignment