Added limit argument for test