Reviewed code, added/removed some TODOS