Fixed the tests and add a badge for the build