Added button to set background, general bug fixing