move code into /src folder