Fix copy and paste code from internet