Edit the UI to allow for everything in the Homework Assignment 1 to fit!