Added button and changes to the template and controller