added docs + cleanups