My Python coursework from year 1