Updated readme + fixed a bug with NaN