adding basic code that got the job done