Remove null returns & null checks