Task No.2 for Optimus Hackathon by Atul Mittal