Updated example file. Best attempt at unit test so far