:bear: Add tests for functionality