Moving unit tests to tests folder