cleaned up / added documentation