Re-adding main
Fixing path in test