Considering its free, this app does a decent job of tracking your sleep stages. If u wake in the night, just pick up your phone, then it will note u were awake. Otherwise, if u are awake, but laying still, it will "think" you are sleeping. Not perfect, but better than nothing.
lisaeawatson about Sleep Meister Lite, v1.10.15