Add tests + comments