Revert "Revert "Added some lines""

This reverts commit 18793933702634a9b9f7eaa51d272845a5a27cee.