added some buttons and functionality