Updated with some stuff

Fixed problems with third button