Fix typo
Add simple tests
Add base function and pass all tests