Changes worked... undoing test and adding correct change