Code clean up + added some code docs.