Adding test to src directory for easier testing