Small change, button was in wrong place