Improve script for processing the BEL small/large corpus