Add return and update homepage