Python first program testing