finshed adding tests