dice gambling script

from random import randint

def diceRoll(userGuess, numRoll):
  if userGuess == numRoll:
    return True
  else:
    return False
    
def main():
  numRoll = randint(1, 6)
  userGuess = int(input("Guess a number from 1 to 6: "))
  
  if diceRoll(userGuess, numRoll):
    print("You win!")
  else:
    print("You lose. Better luck next time!")
    
main()