Removed hard coded height of 320px