Added a button and some other shit!