Added separate rules for test and production code