Made small changes to UDJViewController