Assignment 2 minor UI changes