added button functionality sort of