optimise code
optimise code