Implemented the second scenario
Created unit factory to simplify the creation of units