Initial Commit

CSE Group coding test. Original file was bit sloppy (pressure!) so, as practice for myself, rewrote the script.