Minor Bug fixes..