fixing bug in main call script for testing nn