Slight Changes/Optimisation