Added inf norm and two norm functionality