Fixed indentation
Bringing the button back for now