fixing indentation issue
fixed bad function call, changed comprehension