Yea I think the level design is probably the most difficult part for this kind of puzzle type game. Don't sweat about it though; it s a great job for your first jam - you completed a solid game in short time!
