adding code from DB class
Including only my own code for prog1