code optimisisation

code optimisisation