Added 1 line for memory management