Cleaning up.
Added missing use statement
Removed unneeded new line