Range method modified

List Comprehension used