Small fixes in program.