improving buttons, not sure why this was hard?