Cleaned Code (Formatted and Removed Unused Imports)