Addition of code folder and test.py