added drawing to main.cpp
updated draw in main.cpp