Budget expectations likely to keep stocks in green zone