Just moved the code to src folder