Update TODO
Handle nulls