Added test files to check on functionality and start TDD