made changes to test classed