Added comments and solution array