I think this could definitely be expanded on further! I was really surprised it ended so soon, as I felt you could make a lot more interesting levels with the mechanics.
That part of the last level where you have to jump in time, otherwise reset the whole level, might be a bit too punishing, however.