Minor Changes in Code