Marked main and test as python packages and written first test