Fixing the Add Media button UI