Add difficulty function, returns number

Makes it more fun!