added output option, normalize probability