Update trab1.py