Simple graph

Improve pythonicity and readability