Add test
Better comment