[#5] Begin adding unit tests