Updated vdp_collocation2.py