Relates to #6. Made the button conditional using a controller match