Update the code with grid

[Dinishika98]

Purpose / Goal

Please specify here what you're planning to achieve by this pull request and how it can help in regard of this work-space.

• mention the issue number, if exists.
• add some video/screenshot highlighting your changes on GUI.

Type

Please mention the type of PR

• [ ] Bug Fix
• [ ] Refactoring / Technology upgrade
• [ ] New Feature
• [ ] Documentation
• [ ] Other : | Please Specify |

Note : Please ensure that you've read contribution guidelines before raising this PR.