Making slight changes to __init__ method for Unit
Fixed some indentation issues in Unit's __init__
Test files for imports