Add unit tests for exercise

This pull request changes some of the exercises to use main as target function and testing point. Change some formatting, and add python tests.