Revert "Revert "add 3 more lines""

This reverts commit 082cde657ae34c321f9f877a1060e2ccb28d114f.