Sample Code for Interview assignment