Cleanup Code
Added doc strings