Moved class to separate file and included it here.