Tests now passing locally. Closes #5.