Fix tests
Add badges in readme