Como colocar o cursor no inputText quando carregar um form jsf?

<?xml version="1.0" encoding="ISO-8859-1"?>
<ui:composition xmlns="http://www.w3.org/1999/xhtml"
	xmlns:ui="http://java.sun.com/jsf/facelets"
	xmlns:h="http://java.sun.com/jsf/html"
	xmlns:f="http://java.sun.com/jsf/core"
	xmlns:a4j="http://richfaces.org/a4j"
	xmlns:rich="http://richfaces.org/rich"
	xmlns:c="http://java.sun.com/jstl/core"
	template="/templates/common.xhtml" >
	<ui:define name="body">	
	<h:form id="form2" >
		
		<rich:panel header="Busca">
			<h:outputText value="Entre com o código: " />
			<h:inputText  styleClass="teste" id="codigo" binding="#{produtoMB.codigo}">		
				<rich:jQuery selector="#codigo"  query="focus()" timing="onload" />
			</h:inputText>
			<rich:hotKey selector="#codigo" key="return" 
                            handler="#{rich:element('busca')}.onclick();return false;"/>
			<a4j:commandButton action="#{produtoMB.buscaId}" value="Buscar" 
			    reRender="output,form2" id="busca"/>
		</rich:panel>
	</h:form>

		<h:panelGrid id="output" columns="2">
			<h:outputText value="Codigo: " />
				<h:outputText value="#{produtoMB.produto.codigo}" />
			<h:outputText value="Preço: " />
				<h:outputText value="#{produtoMB.produto.preco}" />
			<h:outputText value="Descrição: " />
				<h:outputText value="#{produtoMB.produto.descricao}" />
		</h:panelGrid>
		
	</ui:define>
</ui:composition>