Fix minor UI issues of project 2