added badge, reduced complexity