Added import statements for UI classes