Rework get_sig
Change numpy import for less clutter