Tidy up
Use math.random instead of math.rand