added optional number state offset to fock_distribution visualization function