fixed short code code, it now returns rather than echos