set key, and button image
Clean up and some styling