Test python code for school