tried to fix the button at the end