Removing jaxy, fixes #1