Adds badge to README + cleanup