Moved code into src folder