moved code to the src folder