Code separation into 2 different classes.