This pretty much nails it, well done.
Maybe a bit unrealistic to have complete agency over when to go to sleep, I might have to add some house rules for that :)
Viewing post in My Brain Is A Stick Of Butter comments
Perhaps when you want to try and sleep roll 1d6, on a 1 you fail to sleep, mark time you spent tossing and turning being frustrated, on a 2-5 you fall asleep, on a 6 you have a brilliant idea, you don't fall asleep but you do manage to make 2 progress on a project, draw a card to find out which one?