Trying to fix the buttons again :)