minor design changes. someone help me :)