Add test files

Add TODO notes