Used list comprehention on chris911's function

Used list comprehention on chris911's function