Refactor task proxy

Remove task proxy class generation.