added paper suggested from Kingma