Moved code to src directory