BalajiN | #49 | Adding unit tests