Button now does what you expect it to do