Lab project in python