Updating testing file with protocols so it will work as a demo