little design corrections