Added initial unit tests (#15)