fixing basic code
fixing basic code