Improve logic and doc strings
Refactor task proxy

Remove task proxy class generation.