changes to CW1 Q1-3

more efficient version for Q1-3