Minor code changes
Drops in codeception