Unit tests

[AkshitaB]

• To be added as and when problems are encountered.
• Not aiming for 100% coverage, just more than what we have now.