Fixed bug in return statement. now marks 'initialized' correctly