Made the UI Prettier

- fixed a bug that made it ugly