Nerual Network code clean up complete