Finally found the answer thanks to this post :]
Seems like I have to construct the CustomDisplay class outside the function with static keyword.
How to redraw the object created by customdisplay? - Grasshopper Developer - McNeel Forum
1 Like