Working on integrating controller with other classes