Add .rb && minor modifications