Allign code with docs