added generator into the system