Simple dice rolling program in python