Add noise to sampling visual unit. Add code to visualise reconstruction of test data