Removed unused import from classes related to UI