finished up adding tests