Refactor propagate, change check dims