Removed unnecessary imports on UI classes