This fixes #1, as well as badges in the readme