Minor changes to solution code