Edited testing class