Updated function signature
Updated dist JS