Create tests directory and move tests from src