Added button and (unfinished) menu funtionality