Updated the UI and made it look really, really good