Create draw_heart.py