Added improved tests for after_id & before_id