updates test and creates class for generate