Added additional tests for IO and keygen