Cute, stylish and functional, but needs a little refactoring