Added quit button (needs a little tweaking)