Added drawing engine
Update the code with the drawing engine.