Add return type to methods