This solution is working in my local intellij setup but not with Databricks notebook setup.