Added the cancel button, have to hook it up