added modularity
began stage_b