Add button functions

Better save function, hover buttons, etc.