Updated Unit Tests for Omar Lalani