change to use a list comprenh
remove parameter