Code re-factoring 1.2