adding some stuff to controller