Running SSD, updated gradient code