Updated files to add some functionality.