added arguments and updated rank for train()
cleanup dependencies
updated launch instructions