Some more UI stuff. Added cancel button