Updating test classes