Fixed a bug
added some buttons