Update method
Fixing docblocks / use statements.