Removed unused import and formated code