Added more write back stuff