Modify to match testing scenario as presented on MKS Coding Challenge webpage