Add in nicer buttons