Slight modification of .rb files