When my baby was 3 mo, she loves to be put on her tummy gym mattress and looking at the toys hanging on top on her. When she woke up, usually the most she will stay awake for 1.5-2hours then go for 2 short naps and one long nap at around 2pm (can be up to 3hours)