small readme update
Fixed plot
Updated paper