A modification of code from a SWC Bootcamp