Added functionality for adding noise into the system