Heat’s Dion Waiters says it felt good to be back on court