Removed useless import
Writing unit tests