Add tests & fix CS