added 2 more buttons for functionality