Changed code to reflect changes in database