Added button styling and cleaned up code