Code re-factoring 1.5