Initial Commit
Add test code to main

*should be in a test case