Adding height restriction