complex shape v2.0

I added an if else statement to the project in order to have the
assignment up to spec