Revert "Removed some lines of code"

This reverts commit dea94cb29dd6361efb37925d406da6404b9fc2df.