[minor] Added more tests
[minor] Sped up computing exponential