Bug fixes :---DDDD