Added function for calculating disc size