Need to hook up return registers for functions...