Bug fixes. Check for null values