screening test python (first time)