refractor - add controllers