Remove extra line
Turn rand generator into a class