Added buttons and made it prettier.