Add Null checks