Improve ruby style