Changes made after small demo to prof