Expand unit tests of application

[IsaacMorris1980]

Need to expand the unit tests to protect the app from regression

[IsaacMorris1980]

Will be easier when #78 is completed