Shape up in two months or go, Gordon Brown warned