modified definition for good behavior with relu