Yes, collapsing specific replies is a bit of a limitation at the moment, but moving between comments with j/k is something that can be implemented!