Added buttons and formatted code