Fixed bug where clicking add button too soon made image button not reappear.