make .py same as notebook