Program to find average marks in Python