Still not perfect, but styles clear now on button press