Small design changes on Math project