A modified Villain formulation of fractons and other exotic theories