Assignment 2 extra credit: improved UI