cleaning up + adding documentation