Updated UI to work with this week's assignment