Lab 3 python