Fixing small error so location now persists
Adding classes to button properly.