updated highlighted square visual appearance
moved color generation so that it triggers before it's needed
added basic code for saving a puzzle (commented out)