Removed unused use statement
Added class reference
Another try