add badges to README
cleanup code