Tried to make the comments more numpy