La til lab2.py, ikke helt ferdig