Put existing code into new function `in_window` because we also want
to use it in another script: `plot_traj.py`.