Fix tests and added badge