Cleaning up image button code