REVIEW CODE

Added code chris explained need to check it works correctly