Changed file to handle new src directory