Refactored tests into functinal and unit.
Removed print statement and redundant import.