Added transitions to match class lesson