Added reference and changed minor stuff