code works now