More computer exercise in python