draw equation using python