Refactor RF tests

Try to minimize random failures by waiting for compose screen to load