Updated documentation + bug fixes