The python code I wrote as part of the assignments for the WP2 experiments towards the dress rehearsal