Replaced hardcoded classes with values from parameters to be able to override these classes.