Handle routing better. Add in view class.