Add tests covering the functionality implemented so far