Derpe earlier.. added stuff to clean_up function... this could use work...
now to add button?