Finished extra task for Assignment 4.