Address PR comments
https://github.com/RtsAiResearch/CaseVisualizer/pull/14

Signed-off-by: MHesham <mhesham.fcis@gmail.com>