Updated readme, code optimisation