Styling issues

@puneet0191 
I have removed this function right now because it is causing error.
I will do a separate PR to add this function when the error is resolved