UI functionality done, just need to "look good" now