Minor changes. Not displaying pieces correctly