Changed so arg is tuple not np array

Fixed line breaks PEP8