Update tests section to be more understandable