Added python implementation for exercise 02