Add code to visualise rbm reconstruction