Made button a function for more inline code