Moved tests into src