moved code to src folder