Unit Tests for the existing functionality