changed code to include public